package com.nowcoder.community.service;

import com.github.benmanes.caffeine.cache.CacheLoader;
import com.github.benmanes.caffeine.cache.Caffeine;
import com.github.benmanes.caffeine.cache.LoadingCache;
import com.nowcoder.community.dao.DiscussPostMapper;
import com.nowcoder.community.entity.DiscussPost;
import com.nowcoder.community.util.RedisKeyUtil;
import org.checkerframework.checker.nullness.qual.NonNull;
import org.checkerframework.checker.nullness.qual.Nullable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;
import org.springframework.web.util.HtmlUtils;

import javax.annotation.PostConstruct;
import java.util.Date;
import java.util.List;
import java.util.concurrent.TimeUnit;

@Service
public class DiscussPostService {

    private static final Logger logger = LoggerFactory.getLogger(DiscussPostService.class);

    @Autowired
    private DiscussPostMapper discussPostMapper;

    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    @Value("${caffeine.posts.max-size}")
    private int cacheMaxSize;

    @Value("${caffeine.posts.expire-seconds}")
    private int cacheExpireSeconds;

    // 本地缓存Caffeine核心接口: 父接口Cache, LoadingCache(同步), AsyncLoadingCache(异步)
    // 适用于缓存的数据：数据不频繁更新的，数据在一段时间内允许不发生改变，数据被访问得较频繁。（查看热帖）
    // 帖子列表缓存
    private LoadingCache<String, List<DiscussPost>> postListCache;

    // 帖子总数缓存
    private LoadingCache<Integer, Integer> postRowsCache;

    @PostConstruct
    public void init() {
        // 在项目启动的时候执行该方法，只需要初始化一次
        // 初始化 postListCache
        postListCache = Caffeine.newBuilder()
                .maximumSize(cacheMaxSize)
                .expireAfterWrite(cacheExpireSeconds, TimeUnit.SECONDS)
                .build(new CacheLoader<String, List<DiscussPost>>() {
                    // 默认的数据加载实现，当调用get取值的时候，如果在本地缓存中key没有对应的值，就调用这个方法进行加载本地缓存
                    @Override
                    public @Nullable List<DiscussPost> load(@NonNull String s) throws Exception {
                        String[] split = s.split(":");
                        if (split.length != 2) {
                            throw new IllegalArgumentException("参数错误！");
                        }
                        int limit = Integer.parseInt(split[0]);
                        int offset = Integer.parseInt(split[1]);

                        // 从redis中取数据（二级缓存）
                        String hotPostKey = RedisKeyUtil.getHotPostKey(limit, offset);
                        if (redisTemplate.hasKey(hotPostKey) != null && redisTemplate.hasKey(hotPostKey)) {
//                            logger.debug("load discussPosts from redis.");
                            return (List<DiscussPost>) redisTemplate.opsForValue().get(hotPostKey);
                        }

//                        logger.debug("load discussPosts from database.");
                        List<DiscussPost> discussPostList = discussPostMapper.selectDiscussPosts(0, limit, offset, 1);
                        // 加载到redis，有效时间为 5分钟
                        redisTemplate.opsForValue().set(hotPostKey, discussPostList, 5, TimeUnit.MINUTES);
                        return discussPostList;
                    }
                });

        // 初始化 postRowsCache
        postRowsCache = Caffeine.newBuilder()
                .maximumSize(cacheMaxSize)
                .expireAfterWrite(cacheExpireSeconds, TimeUnit.SECONDS)
                .build(new CacheLoader<Integer, Integer>() {
                    // 默认的数据加载实现，当调用get取值的时候，如果在本地缓存中key没有对应的值，就调用这个方法进行加载本地缓存
                    @Override
                    public @Nullable Integer load(@NonNull Integer key) throws Exception {
//                        logger.debug("load discussPostRows from database.");
                        return discussPostMapper.selectDiscussPostRows(key);
                    }
                });
    }

    // 根据userId分页查询帖子，userId为 0 时查询所有帖子，orderMode为 1时是查询热帖
    public List<DiscussPost> findDiscussPosts(int userId, int limit, int offset, int orderMode) {
        if (userId == 0 && orderMode == 1) {
            // 查看热帖时，从缓存中取数据
            return postListCache.get(limit + ":" + offset);
        }

//        logger.debug("load discussPosts from database.");
        return discussPostMapper.selectDiscussPosts(userId, limit, offset, orderMode);
    }

    // 根据userId查询帖子总数，userId为 0 时查询所有帖子
    public int findDiscussPostRows(int userId) {
        // 获得所有帖子数量时，从缓存中取数据
        if (userId == 0) {
            return postRowsCache.get(userId);
        }

//        logger.debug("load discussPostRows from database.");
        return discussPostMapper.selectDiscussPostRows(userId);
    }

    // 新增帖子
    public int addDiscussPost(DiscussPost discussPost) {
        discussPost.setTitle(HtmlUtils.htmlEscape(discussPost.getTitle()));
        discussPost.setContent(HtmlUtils.htmlEscape(discussPost.getContent()));
        return discussPostMapper.insertDiscussPost(discussPost);
    }

    // 根据id查找帖子
    public DiscussPost findDiscussPostById(int id) {
        return discussPostMapper.selectDiscussPostById(id);
    }

    // 更新帖子type
    public int updateType(int id, int type) {
        return discussPostMapper.updateType(id, type);
    }

    // 更新帖子status
    public int updateStatus(int id, int status) {
        return discussPostMapper.updateStatus(id, status);
    }

    // 更新帖子score
    public int updateScore(int id, double score) {
        return discussPostMapper.updateScore(id, score);
    }

}
